#!/bin/bash
module purge
module load  compilers/cuda/11.8 compilers/gcc/9.3.0 cudnn/8.4.0.27_cuda11.x anaconda
source activate torch2.1_cu118
# python -u run.py config/exp1_config.yaml --use_part_label
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128
export OMP_NUM_THREADS=1
#python -u train.py config.yaml
#### for train  
# python -u train-para.py config.yaml
### for test
# python -u test-para.py config.yaml /home/bingxing2/home/scx8ajl/ldz/pressure-related/new-transolver/code/best_model_transover_0.001-mlp.pt
#python -u training_dara_1.py
# python -u test-para.py config.yaml ./ckpt/1-5-3shape-mixloss/Jan05_00-52-10/best_model_transover_max_mae_0.0127568560.pt
python -u test-para.py config.yaml ./best_model_mixloss_0.0585814789.pt